#pragma once
#include <string>
#include <vector>
#include <iostream>
#include <opencv2/opencv.hpp>
#include <stdint.h>

#include <rknn_api.h>
#include "model_struct.h"
#include "base_predictor.h"

namespace ai {

class ModelConfig;
class RKModelManager;

#define MAX_SUPPORT_CHANNEL 4

struct RKPackPredictor : public BasePackPredictor {
  RKPackPredictor(): BasePackPredictor() { }
  std::vector<rknn_input> inputs;
  std::vector<rknn_output> outputs;
  rknn_tensor_format fmt;
  std::vector<int> nchw;
  rknn_context ctx;
};








}   // namespace ai 
